#!/bin/sh

# Copyright 2012-2013 Edward W. Koenig, Vancouver, WA, USA
# All rights reserved.
#
# Redistribution and use of this script, with or without modification, is
# permitted provided that the following conditions are met:
#
# 1. Redistributions of this script must retain the above copyright
#    notice, this list of conditions and the following disclaimer.
#
# THIS SOFTWARE IS PROVIDED BY THE AUTHOR ''AS IS'' AND ANY EXPRESS OR IMPLIED
# WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
# MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO
# EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
# PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
# OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
# WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
# OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
# ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
#
# nvidia-switch utility 2.0.1
#
# A tool to install and cleanly remove the nvidia driver without stomping on the xorg files.
# Note: this tool "should" allow switching on the fly, but why? Pick a driver! You will still 
# need to edit a bunch of other config files

# TO-DO: Bounds/error checking, add more COMPAT32? 

set -e

ROOT="/"
CWD=$(pwd)
LIBSUFFIX="LIBDIRSUFFIX" # This will be replaced in the build script
COMPAT32="LIB32FLAG" # This will be replaced in the build script (yes | no)
INC="${ROOT}usr/include/GL"
LIB="${ROOT}usr/lib${LIBSUFFIX}"
LIB32="${ROOT}usr/lib"
XMOD="${LIB}/xorg/modules"
XLIB="$XMOD/extensions"
NV_VERSION='PKGVERSION' # This will be replaced in the build script
GL_VERSION="1.2"
NV_CONF="/usr/share/X11/xorg.conf.d/10-nvidia.conf"

save_GL(){
  cd "$LIB"
  if [ "$1" = "nvidia" ]; then
    if [ ! -e libGL.so.$GL_VERSION-xorg ]; then
       mv libGL.so.$GL_VERSION libGL.so.$GL_VERSION-xorg
    fi
    ln -sf libGL.so.$NV_VERSION-nvidia libGL.so.$NV_VERSION
    ln -sf libGL.so.$NV_VERSION libGL.so.1
    ln -sf libGL.so.1 libGL.so
# Check for multilib configuration 
    if [ "$COMPAT32" = "yes"  ]; then
      cd $LIB32
      if [ ! -e libGL.so.$GL_VERSION-xorg ]; then
        mv libGL.so.$GL_VERSION libGL.so.$GL_VERSION-xorg
      fi
        ln -sf libGL.so.$NV_VERSION-nvidia libGL.so.$NV_VERSION
        ln -sf libGL.so.$NV_VERSION libGL.so.1
        ln -sf libGL.so.1 libGL.so
    fi
  elif [ "$1" = "xorg" ]; then
    mv libGL.so.$GL_VERSION-xorg libGL.so.$GL_VERSION 
    ln -sf libGL.so.$GL_VERSION  libGL.so.1
    ln -sf libGL.so.1 libGL.so
    rm -f libGL.so.$NV_VERSION
# Check for multilib and reverse any "damage"
    if [ "$COMPAT32" = "yes"  ]; then
      cd $LIB32
      rm -f libGL.so
      rm -f libGL.so.1
      rm -f libGL.so.$NV_VERSION
      if [ -f libGL.so.$GL_VERSION-xorg ]; then
        mv libGL.so.$GL_VERSION-xorg libGL.so.$GL_VERSION
      fi
      ln -sf libGL.so.$GL_VERSION  libGL.so.1
      ln -sf libGL.so.1 libGL.so
    fi
  fi
  cd "$CWD"
}

save_wfb(){
  cd "$XMOD"
  if [ "$1" = "nvidia" ]; then
	ln -sf libnvidia-wfb.so.$NV_VERSION-nvidia libnvidia-wfb.so.$NV_VERSION
	ln -sf libnvidia-wfb.so.$NV_VERSION libnvidia-wfb.so.1
	if [ ! -e libwfb.so ]; then
            mv libwfb.so libwfb.so-xorg
            ln -sf libnvidia-wfb.so.$NV_VERSION libwfb.so
        fi
  elif [ "$1" = "xorg" ]; then
	rm -f libnvidia-wfb.so.$NV_VERSION
	rm -f libnvidia-wfb.so.1
        if [ -L libwfb.so ]; then
	    rm -f libwfb.so
            if [ -e libwfb.so-xorg ]; then
                mv libwfb.so-xorg libwfb.so
            fi
        fi
  fi 
  cd "$CWD"
}

save_glx(){
  cd "$XLIB"
  if [ "$1" = "nvidia" ]; then
	if [ ! -e libglx.so-xorg ]; then
	    mv libglx.la libglx.la-xorg
	    mv libglx.so libglx.so-xorg
	fi
	ln -sf libglx.so.$NV_VERSION-nvidia libglx.so.$NV_VERSION
	ln -sf libglx.so.$NV_VERSION libglx.so
  elif [ "$1" = "xorg" ]; then
	rm -f libglx.so.$NV_VERSION
	rm -f libglx.so
	mv libglx.so-xorg libglx.so
	mv libglx.la-xorg libglx.la
  fi 
cd "$CWD"
}

nvidia_ldconfig(){
   /sbin/ldconfig
   #Generate correct symink for that lib
   /sbin/ldconfig -l $1
}

nvidia_install(){
  echo $'Installing to nvidia-driver files!\n'
  echo "Make sure the nvidia driver is ENABLED in /etc/X11/xorg.conf"
  echo "and in /etc/X11/xorg.conf.d."
  echo "Otherwise, this may lead to improperly working drivers."

  save_GL "nvidia"
  save_wfb "nvidia"
  save_glx "nvidia"
  

  LD_NVIDIA="${LIB}/libGL.so.$NV_VERSION-nvidia"
  nvidia_ldconfig $LD_NVIDIA
}

nvidia_remove(){
  echo $'Returning to stock xorg files!\n'
  echo "Make sure the nvidia driver is DISABLED in /etc/X11/xorg.conf"
  echo "and in /etc/X11/xorg.conf.d."
  echo "Otherwise, this may lead to improperly working drivers."

  save_GL "xorg"
  save_wfb "xorg"
  save_glx "xorg"


  if [ -e $NV_CONF ]; then
	rm -f $NV_CONF
  fi
  
  LD_NVIDIA="${LIB}/libGL.so.$GL_VERSION"
  nvidia_ldconfig $LD_NVIDIA
}

usage(){
  echo "Usage:"
  echo " --install		Set up nvidia driver files"
  echo " --remove		Return to stock xorg files and remove all symlinks"
  echo " --help			Show this help message"
}

if [ "$1" = '--install' ]; then
  nvidia_install
elif [ "$1" = '--remove' ]; then
  nvidia_remove
elif [ "$1" = '--help' ]; then
  usage
else
  usage
fi
